package com.gengzp.mq.common.utils;

import jakarta.annotation.PostConstruct;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.data.redis.core.Cursor;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ScanOptions;
import org.springframework.stereotype.Component;

import java.nio.charset.StandardCharsets;
import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * @ClassName RedisUtils
 * @Description redis工具类
 * @Author gengzp
 * @Date 2025/6/8 15:18
 */
@Component
public class GengzpMqRedisUtils {

    @Autowired
    @Qualifier("gengzpMqRedisTemplate")
    private RedisTemplate<String, Object> redisTemplateInstance;
    private static RedisTemplate<String, Object> redisTemplate;

    private GengzpMqRedisUtils() {
    }

    @PostConstruct
    public void init() {
        redisTemplate = redisTemplateInstance;
    }

    // ===================== 写操作命令 =====================

    /**
     * 将数据保存到redis
     *
     * @param key     键
     * @param value   值
     * @param timeout 过期时间
     * @param unit    时间单位
     */
    public static void set(String key, Object value, long timeout, TimeUnit unit) {
        redisTemplate.opsForValue().set(key, value, timeout, unit);
    }

    /**
     * 将数据添加到队列最右
     *
     * @param key   键
     * @param value 值
     */
    public static void rightPush(String key, Object value) {
        redisTemplate.opsForList().rightPush(key, value);
    }

    /**
     * 删除元素
     *
     * @param key 键
     */
    public static void delete(String key) {
        redisTemplate.delete(key);
    }

    // ===================== 读操作命令 =====================


    /**
     * 取出元素
     *
     * @param key 键
     * @return 值
     */
    public static Object get(String key) {
        return redisTemplate.opsForValue().get(key);
    }

    /**
     * 阻塞获取队列最左侧元素
     *
     * @param key     键
     * @param timeout 最大阻塞时间
     * @param unit    时间单位
     * @return 队列最左侧元素
     */
    public static Object leftPop(String key, long timeout, TimeUnit unit) {
        return redisTemplate.opsForList().leftPop(key, timeout, unit);
    }

    /**
     * 使用 SCAN 命令迭代获取符合条件的键，避免阻塞 Redis 服务器
     *
     * @param pattern 键匹配模式，如 "user:*"
     * @param count   每次迭代的近似数量（非严格限制）
     * @return 匹配的键集合
     */
    public static Set<String> scan(String pattern, long count) {
        Set<String> keys = new HashSet<>();
        ScanOptions options = ScanOptions.scanOptions()
                .match(pattern)
                .count(count)
                .build();

        // 使用 executeWithStickyConnection 确保在同一个连接上执行完整的迭代
        redisTemplate.executeWithStickyConnection(connection -> {
            Cursor<byte[]> cursor = null;
            try {
                cursor = connection.scan(options);
                while (cursor.hasNext()) {
                    keys.add(new String(cursor.next(), StandardCharsets.UTF_8));
                }
            } finally {
                // 确保游标被关闭
                if (cursor != null) {
                    try {
                        cursor.close();
                    } catch (Exception ignored) {
                    }
                }
            }
            return null;
        });

        return keys;
    }

}
